package top.codecrab.lock;

import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicReference;

/**
 * 自旋锁实现方法
 *
 * @author codecrab
 * @since 2021年06月30日 11:36
 */
public class SpinLockDemo {

    //默认没有赋值就是null
    private final AtomicReference<Thread> atomicReference = new AtomicReference<>();

    /**
     * 加锁
     * 实现原理：利用原子类的CAS思想，如果赋值失败，就一直自旋重试，
     * 如果赋值成功，就会返回true，结束自旋，否则一直循环比较，直到解锁方法被调用
     */
    public void lock() {
        System.out.println(Thread.currentThread().getName() + "进入加锁方法。。。");
        Thread thread = Thread.currentThread();

        //如果赋值成功，就会返回true，结束自旋，否则一直循环比较，直到解锁方法被调用
        while (!atomicReference.compareAndSet(null, thread)) {

        }
    }

    /**
     * 解锁方法原理：利用当前线程，去比较并更新原子类的值，变为null，让在循环的线程成功比较并更新原子类的值，结束循环。解锁成功
     */
    public void unlock() {
        System.out.println(Thread.currentThread().getName() + "进入解锁方法。。。");
        Thread thread = Thread.currentThread();
        atomicReference.compareAndSet(thread, null);
        System.out.println(Thread.currentThread().getName() + "解锁成功。。。");
    }

    public static void main(String[] args) {
        SpinLockDemo spinLock = new SpinLockDemo();

        new Thread(() -> {
            spinLock.lock();
            try {
                TimeUnit.SECONDS.sleep(5);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            spinLock.unlock();
        }, "AA").start();

        try {
            TimeUnit.SECONDS.sleep(1);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        new Thread(() -> {
            spinLock.lock();
            spinLock.unlock();
        }, "BB").start();
    }
}
